package com.web;

import lombok.SneakyThrows;

import javax.crypto.Cipher;
import javax.crypto.SecretKey;
import javax.crypto.SecretKeyFactory;
import javax.crypto.spec.DESKeySpec;
import java.security.MessageDigest;
import java.security.SecureRandom;
import java.util.Base64;

/**
 * @author：zhang guoxiang
 * @date：2022/5/5
 */
public abstract class EncryptUtils {


    /**
     * MD5加密
     *
     * @param content 加密内容
     * @return 返回加密数据
     */
    @SneakyThrows
    public static String md5Encrypt(final Object content) {
        if (content == null || "".equals(content)) {
            return "";
        }
        MessageDigest messageDigest = MessageDigest.getInstance("MD5");
        byte[] digest = messageDigest.digest(content.toString().getBytes());
        return new String(digest);
    }

    /**
     * base64 加密
     *
     * @param content 加密内容
     * @return 返回加密
     */
    public static String base64Encrypt(final Object content) {
        if (content == null || "".equals(content)) {
            return "";
        }
        return Base64.getEncoder().encodeToString(content.toString().getBytes());
    }

    /**
     * base64 加密
     *
     * @param content 加密内容
     * @return 返回加密
     */
    public static byte[] base64EncryptToByte(final Object content) {
        if (content == null || "".equals(content)) {
            return new byte[0];
        }
        return Base64.getEncoder().encode(content.toString().getBytes());
    }

    /**
     * base64 解密
     *
     * @param content 解密内容
     * @return 返回解密内容
     */
    public static byte[] base64Decrypt(final String content) {
        if (content == null || "".equals(content)) {
            return new byte[0];
        }
        return Base64.getDecoder().decode(content.getBytes());
    }

    /**
     * des对称加密
     *
     * @param key     加密使用的key
     * @param content 需要加密的内容
     * @return 返回加密的内容
     */
    @SneakyThrows
    public static byte[] desEncrypt(String key, byte[] content) {
        SecureRandom random = new SecureRandom();
        DESKeySpec desKey = new DESKeySpec(key.getBytes());
        //创建一个密匙工厂，然后用它把DESKeySpec转换成
        SecretKeyFactory keyFactory = SecretKeyFactory.getInstance("DES");
        SecretKey securekey = keyFactory.generateSecret(desKey);
        //Cipher对象实际完成加密操作
        Cipher cipher = Cipher.getInstance("DES");
        //用密匙初始化Cipher对象
        cipher.init(Cipher.ENCRYPT_MODE, securekey, random);
        //现在，获取数据并加密
        //正式执行加密操作
        return cipher.doFinal(content);
    }

    /**
     * des 对称解密
     *
     * @param content 解密内容
     * @param key     加密使用key
     * @return 返回解密后的内容
     */
    @SneakyThrows
    public static byte[] desDecrypt(byte[] content, String key) {
        // DES算法要求有一个可信任的随机数源
        SecureRandom random = new SecureRandom();
        // 创建一个DESKeySpec对象
        DESKeySpec desKey = new DESKeySpec(key.getBytes());
        // 创建一个密匙工厂
        SecretKeyFactory keyFactory = SecretKeyFactory.getInstance("DES");
        // 将DESKeySpec对象转换成SecretKey对象
        SecretKey securekey = keyFactory.generateSecret(desKey);
        // Cipher对象实际完成解密操作
        Cipher cipher = Cipher.getInstance("DES");
        // 用密匙初始化Cipher对象
        cipher.init(Cipher.DECRYPT_MODE, securekey, random);
        // 真正开始解密操作
        return cipher.doFinal(content);
    }
}
